import os
import shutil

def extract_files(input_path, output_path):
    # 创建输出路径
    if not os.path.exists(output_path):
        os.makedirs(output_path)

    # 遍历输入路径下的所有文件
    files = sorted(os.listdir(input_path))  # 获取输入路径下的所有文件，并按字母顺序排序
    for i, file in enumerate(files):
        if i % 5 == 0:  # 每隔{}个文件抽取一个文件
            file_path = os.path.join(input_path, file)
            if os.path.isfile(file_path):  # 确保是文件而不是文件夹
                # 复制文件到输出路径
                # shutil.copy2(file_path, output_path)

                #移动文件到输出路径
                shutil.move(file_path, output_path)
                print(f"Extracted file: {file}")

# 输入路径和输出路径
input_path = "/media/jolly/HIK/Brain Tumor Segmentation Dataset/label/predictResult/"
output_path = "/media/jolly/HIK/Brain Tumor Segmentation Dataset/label/predictResult1/"

# 调用函数抽取并复制文件
extract_files(input_path, output_path)